package example.demo.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * 交换机确认回调配置,以及继承了回退接口
 */
@Component
@Slf4j
public class MyCallBack implements RabbitTemplate.ConfirmCallback,RabbitTemplate.ReturnsCallback {

    // 注入
    @Autowired
    private RabbitTemplate rabbitTemplate;
    // 将当前类注入到 确认回调类中 。。。 当前类只是实现类 ，不注入 无法调用
    @PostConstruct
    public void init(){
        rabbitTemplate.setConfirmCallback(this); // 交换机接受不到消息 回退接口
        /**
         * true：
         * 交换机无法将消息进行路由时，会将该消息返回给生产者
         * false：
         * 如果发现消息无法进行路由，则直接丢弃
         */
        rabbitTemplate.setMandatory(true);
        // 必须要设置setMandatory
        rabbitTemplate.setReturnsCallback(this); // 队列接受不到消息，回退接口
    }

    /**
     * 交换机确认回调方法
     * 1. 发消息 交换机接收到了 回调
     *  1.1 correlationData  保存回调消息的ID及相关信息
     *  1.2 交换机接收到消息 true
     *  1.3 cause null
     * 2. 发消息 交换机接收失败了 回调
     *  2.1 correlationData 保存回调消息的ID及相关信息
     *  2.2 交换机收到消息 ack = false;
     *  2.3 失败的原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        //
        String id = correlationData != null ?correlationData.getId() : "";
        if ( ack ) {
            log.info("交换机已经收到ID为：{}的消息",id);
        } else {
            log.info("交换机还未收到id为{}的消息，由于原因：{}",id,cause);
        }
    }

    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        log.info("消息{}，被交换机{}退回，退回的原因：{}，路由key是多少：{}",
                new Message(message.getBody()),
                exchange,
                replyText,
                routingKey);
    }

    @Override
    public void returnedMessage(ReturnedMessage returned) {
        log.info("消息{}，被交换机{}退回，退回的原因：{}，路由key是多少：{}",
                new Message(returned.getMessage().getBody()),
                returned.getExchange(),
                returned.getReplyText(),
                returned.getRoutingKey());
    }


    /**
     * 消息队列接受消息失败，回退接口、
     * 可以在当消息传递g过程中不可达目的d地时j将消息返回给生产者
     * 只有在不可达目的地时才可以回退
     */

    /*@Override
    public void returnedMessage(ReturnedMessage returned) {
        log.info("消息{}，被交换机{}退回，退回的原因：{}，路由key是多少：{}",
                new Message(returned.getMessage().getBody()),
                returned.getExchange(),
                returned.getReplyText(),
                returned.getRoutingKey());
    }*/

}
